package com.googlecode.familytreesimulator;

import java.util.ArrayList;
import java.util.List;

/** Create GeneDatabase to store all possible
 *  genes
 *
 */
public class GeneDatabase {
	
	List<String> eyeColor = new ArrayList<String>();
	List<String> height = new ArrayList<String>();
	List<String> hairColor = new ArrayList<String>();
	List<String> hairType = new ArrayList<String>();
	List<String> skinTone = new ArrayList<String>();
	
	public GeneDatabase(){
		//List of eye colors
		eyeColor.add("black");
		eyeColor.add("brown");
		eyeColor.add("blue");
		eyeColor.add("green");
		eyeColor.add("hazel");
		
		//List of heights
		height.add("fairly short");
		height.add("average");
		height.add("tall");
		
		/*List of hair colors*/
		hairColor.add("black");
		hairColor.add("brown");
		hairColor.add("blonde");
		hairColor.add("red");
		
		/*List of hair types*/
		hairType.add("straight");
		hairType.add("wavy");
		hairType.add("curly");
		
		/*List of skin tones*/
		skinTone.add("dark");
		skinTone.add("light");
		skinTone.add("tan");
		skinTone.add("olive");
	}

	/** getGeneList(int listNum) gets the corresponding
	 *  gene list
	 * 
	 * @param listNum
	 * @return
	 */
	public List<String> getGeneList(int listNum){
		switch(listNum){
			case 0: return eyeColor;
			case 1: return height;
			case 2: return hairColor;
			case 3: return hairType;
			case 4: return skinTone;
			default: System.out.println("Error, list not found.");
					 return null;
		}
	}
}
